package com.jd.dl.util;

import java.io.*; 
import java.util.zip.*; 
/** 
*功能:zip压缩、解压 
*说明:本程序通过ZipOutputStream和ZipInputStream实现了zip压缩和解压功能. 
*问题:由于java.util.zip包并不支持汉字,当zip文件中有名字为中文的文件时, 
*     就会出现异常:"Exception  in thread "main " java.lang.IllegalArgumentException  
*               at   java.util.zip.ZipInputStream.getUTF8String(ZipInputStream.java:285) 
*解决: 
*　　方法1、修改import java.util.zip.ZipInputStream和ZipOutputStream. 
*          java.util.zip只支持UTF-8,Ant里面可以指定编码. 
*　　方法2、使用Apache Ant里提供的zip工具。 
*          不使用java.util.zip的包,把ant.jar放到classpath中. 
*          程序中使用import org.apache.tools.zip.*; 
* 
*仅供编程学习参考. 
* 
*@author Winty 
*@date   2008-8-3 
*@Usage: 
*   压缩:java Zip -zip "directoryName" 
*   解压:java Zip -unzip "fileName.zip" 
*/ 

public class Zip{ 
    private ZipInputStream  zipIn;      //解压Zip  
    private ZipOutputStream zipOut;     //压缩Zip  
    private ZipEntry        zipEntry; 
    private static int      bufSize;    //size of bytes 
    private byte[]          buf; 
    private int             readedBytes; 
     
    public Zip(){ 
        this(512); 
    } 

    public Zip(int bufSize){ 
        this.bufSize = bufSize; 
        this.buf = new byte[this.bufSize]; 
    } 
     
    //压缩文件夹内的文件  
    /**
     * @param zipDirectory: zip dir full path
     */
    public void doZip(String zipDirectory){ 
        File file; 
        File zipDir; 

        zipDir = new File(zipDirectory); 

        //generate the zip file name
        String zipFileName = zipDir.getName() + ".zip"; 

        try{ 
            this.zipOut = new ZipOutputStream(new BufferedOutputStream(new FileOutputStream(zipFileName))); 
            handleDir(zipDir , this.zipOut); 
            this.zipOut.close(); 
        }catch(IOException ioe){ 
            ioe.printStackTrace(); 
        } 
    } 

    //由doZip调用,递归完成目录文件读取 
    private void handleDir(File dir , ZipOutputStream zipOut)throws IOException{ 
        FileInputStream fileIn; 
        File[] files; 

        files = dir.listFiles(); 
     
        if(files.length == 0){//如果目录为空,则单独创建之. 
        	//ZipEntry的isDirectory()方法中,目录以"/"结尾. 
            this.zipOut.putNextEntry(new ZipEntry(dir.toString() + "/")); 
            this.zipOut.closeEntry(); 
        } else { //如果目录不为空,则分别处理目录和文件.  
            for(File fileName : files){ 
                //System.out.println(fileName); 

                if (fileName.isDirectory()) {
                    handleDir(fileName , this.zipOut); 
                } else {
                    fileIn = new FileInputStream(fileName); 
                    this.zipOut.putNextEntry(new ZipEntry(fileName.toString())); 

                    while((this.readedBytes = fileIn.read(this.buf))>0){ 
                        this.zipOut.write(this.buf , 0 , this.readedBytes); 
                    } 

                    this.zipOut.closeEntry(); 
                }
            }
        }
    }

    /**
     * @param unZipfileName: zip file full path
     */
    public void unZip(String unZipfileName){ 
        FileOutputStream fileOut; 
        File file; 

        try{
        	//make dest dir for unzip files
        	String firstName = unZipfileName.substring(0, unZipfileName.lastIndexOf("."));
        	File unzipDir = new File(firstName);
        	unzipDir.mkdirs();
        	
            this.zipIn = new ZipInputStream (new BufferedInputStream(new FileInputStream(unZipfileName))); 

            while((this.zipEntry = this.zipIn.getNextEntry()) != null){ 
                file = new File(this.zipEntry.getName()); 

                if(this.zipEntry.isDirectory()){ 
                    file.mkdirs(); 
                } 
                else{ 
                    //make parent dir for entry if it's not exist
                    File parent = file.getParentFile(); 
                    if (parent != null) {
                    	if (!parent.exists()) {
	                    	parent.mkdirs();
                    	}
                    }

                    //make dest file
                    File dest = new File (unzipDir.getPath() + File.separator + file.getName());
                    
                    //write content to dest file
                    fileOut = new FileOutputStream(dest);
                    while(( this.readedBytes = this.zipIn.read(this.buf) ) > 0){
                        fileOut.write(this.buf , 0 , this.readedBytes );
                    }
                    fileOut.close();
                }
                this.zipIn.closeEntry();
            }
        }catch(IOException ioe){
            ioe.printStackTrace();
        }
    }

    //set buffer size
    public void setBufSize(int bufSize){
        this.bufSize = bufSize;
    }

}